package com.yuandian.utils.poi;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.URLEncoder;
import java.util.List;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;


/**
 * 样包下载
 *
 * @author User
 */
public class WordUtil {
    private static Logger logger = LoggerFactory.getLogger(WordUtil.class);
    /**
     * 用于控制文件打包下载，避免文件重复
     */
    public static int count = 1;

    /**
     * 下载多个文件
     *
     * @param filePath   filePath
     * @param jsonString jsonString
     * @param fileName   fileName
     * @return File
     */
    public static File ecportFileData2(String filePath, String jsonString, String fileName) {
        // 标记文件生成是否成功
        File file = null;
        try {
            file = new File(filePath + "yangbao");
            if (!file.exists()) {
                file.mkdir();
            }
            file = new File(filePath + "yangbao/" + fileName + ".txt");
            if (!file.exists()) {
                file.createNewFile();
            } else {
                file = new File(file.getName().split("\\.")[0] + "(" + count + ").txt");
                count++;
            }

            // 格式化json字符串
            jsonString = formatJson(jsonString);
            // 将格式化后的字符串写入文件
            Writer write = new OutputStreamWriter(new FileOutputStream(file), "UTF-8");
            write.write(jsonString);
            write.flush();
            write.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
        // 返回是否成功的标记
        return file;
    }

    //导出数据
    public static boolean ecportFileData(HttpServletResponse response, String jsonString, String fileName) {
        logger.info("开始下载样包");
        // 标记文件生成是否成功
        boolean flag = true;

        // 生成json格式文件
        try {
            String fileName1 = URLEncoder.encode(fileName + ".txt", "UTF-8");
            String headStr = "attachment;filename=\"" + fileName1 + "\"";
            response.setHeader("Content-Disposition", headStr);
            // 格式化json字符串
            jsonString = formatJson(jsonString);
            // 将格式化后的字符串写入文件
            Writer write = new OutputStreamWriter(response.getOutputStream(), "UTF-8");
            write.write(jsonString);
            write.flush();
            write.close();
        } catch (Exception e) {
            flag = false;
            e.printStackTrace();
        }
        // 返回是否成功的标记
        return flag;
    }

    //导出数据
    public static boolean ecportFileDataTwo(HttpServletRequest request, HttpServletResponse response, String jsonString, String fileName) {
        logger.info("开始下载样包");
        // 标记文件生成是否成功
        boolean flag = true;

        // 生成json格式文件
        try {
            String fileName1 = "";
            if (request.getHeader("User-Agent").toUpperCase().indexOf("MSIE") > 0) {
                fileName1 = URLEncoder.encode(fileName + ".txt", "UTF-8");
            } else {
                fileName1 = new String((fileName + ".txt").getBytes("UTF-8"), "ISO8859-1");
            }
            String headStr = "attachment;filename=\"" + fileName1 + "\"";
            response.setHeader("Content-Disposition", headStr);
            // 格式化json字符串
            jsonString = formatJson(jsonString);
            // 将格式化后的字符串写入文件
            Writer write = new OutputStreamWriter(response.getOutputStream(), "UTF-8");
            write.write(jsonString);
            write.flush();
            write.close();
        } catch (Exception e) {
            flag = false;
            e.printStackTrace();
        }
        // 返回是否成功的标记
        return flag;
    }

    /**
     * 单位缩进字符串。
     */
    private static String SPACE = "   ";

    /**
     * 返回格式化JSON字符串。
     *
     * @param json 未格式化的JSON字符串。
     * @return 格式化的JSON字符串。
     */
    public static String formatJson(String json) {
        StringBuffer result = new StringBuffer();

        int length = json.length();
        int number = 0;
        char key = 0;

        // 遍历输入字符串。
        for (int i = 0; i < length; i++) {
            // 1、获取当前字符。
            key = json.charAt(i);

            // 2、如果当前字符是前方括号、前花括号做如下处理：
            if ((key == '[') || (key == '{')) {
                // （1）如果前面还有字符，并且字符为“：”，打印：换行和缩进字符字符串。
                if ((i - 1 > 0) && (json.charAt(i - 1) == ':')) {
                    result.append("\r\n");
                    result.append(indent(number));
                }

                // （2）打印：当前字符。
                result.append(key);

                // （3）前方括号、前花括号，的后面必须换行。打印：换行。
                result.append("\r\n");

                // （4）每出现一次前方括号、前花括号；缩进次数增加一次。打印：新行缩进。
                number++;
                result.append(indent(number));

                // （5）进行下一次循环。
                continue;
            }

            // 3、如果当前字符是后方括号、后花括号做如下处理：
            if ((key == ']') || (key == '}')) {
                // （1）后方括号、后花括号，的前面必须换行。打印：换行。
                result.append("\r\n");

                // （2）每出现一次后方括号、后花括号；缩进次数减少一次。打印：缩进。
                number--;
                result.append(indent(number));

                // （3）打印：当前字符。
                result.append(key);

                // （4）如果当前字符后面还有字符，并且字符不为“，”，打印：换行。
                if (((i + 1) < length) && (json.charAt(i + 1) != ',')) {
                    result.append("\r\n");
                }

                // （5）继续下一次循环。
                continue;
            }

            // 4、如果当前字符是逗号。逗号后面换行，并缩进，不改变缩进次数。
            if ((key == ',')) {
                result.append(key);
                result.append("\r\n");
                result.append(indent(number));
                continue;
            }

            // 5、打印：当前字符。
            result.append(key);
        }

        return result.toString();
    }

    /**
     * 返回指定次数的缩进字符串。每一次缩进三个空格，即SPACE。
     *
     * @param number 缩进次数。
     * @return 指定缩进次数的字符串。
     */
    private static String indent(int number) {
        StringBuffer result = new StringBuffer();
        for (int i = 0; i < number; i++) {
            result.append(SPACE);
        }
        return result.toString();
    }

    /**
     * 以流的形式形式下载PDF文件
     * @param file
     * @param response
     * @return
     */
    public static HttpServletResponse downloadPdf(File file,
                                                  HttpServletResponse response) {
        try {
            // 以流的形式下载文件。
            InputStream fis = new BufferedInputStream(new FileInputStream(
                    file.getPath()));
            byte[] buffer = new byte[fis.available()];
            fis.read(buffer);
            fis.close();
            // 清空response
            response.reset();

            OutputStream toClient = new BufferedOutputStream(
                    response.getOutputStream());
            response.setContentType("application/octet-stream");

            // 如果输出的是中文名的文件，在此处就要用URLEncoder.encode方法进行处理
            response.setHeader("Content-Disposition", "attachment;filename="
                    + URLEncoder.encode(file.getName(), "UTF-8"));
            toClient.write(buffer);
            toClient.flush();
            toClient.close();
        } catch (IOException ex) {
            ex.printStackTrace();
        }
        return response;
    }

    /**
     * 文件打包下载
     * files 这个集合就是你想要打包的所有文件， 这里假设已经准备好了所要打包的文件
     */
    public static HttpServletResponse downLoadFiles(String filePath, List<File> files, HttpServletResponse response) throws Exception {
        try {
            /**
             * 创建一个临时压缩文件， 我们会把文件流全部注入到这个文件中 这里的文件你可以自定义是.rar还是.zip
             */
            File file = new File(filePath + "原始数据包.zip");
            if (!file.exists()) {
                file.createNewFile();
            }
            response.reset();
            // response.getWriter()
            // 创建文件输出流
            FileOutputStream fous = new FileOutputStream(file);
            /**
             * 打包的方法我们会用到ZipOutputStream这样一个输出流, 所以这里我们把输出流转换一下
             */
            ZipOutputStream zipOut = new ZipOutputStream(fous);

            /**
             * 这个方法接受的就是一个所要打包文件的集合， 还有一个ZipOutputStream
             */
            zipFile(files, zipOut);
            zipOut.close();
            fous.close();
            return downloadZip(file, response);
        } catch (Exception e) {
            e.printStackTrace();
        }
        /**
         * 直到文件的打包已经成功了， 文件的打包过程被我封装在FileUtil.zipFile这个静态方法中，
         * 稍后会呈现出来，接下来的就是往客户端写数据了
         */

        return response;
    }

    /**
     * 文件打包下载
     * files 这个集合就是你想要打包的所有文件， 这里假设已经准备好了所要打包的文件
     */
    public static HttpServletResponse downLoadFiles(List<File> files,
                                                    HttpServletRequest request, HttpServletResponse response)
            throws Exception {
        try {
            String filePath = System.getProperty("java.io.tmpdir");
            /**
             * 创建一个临时压缩文件， 我们会把文件流全部注入到这个文件中 这里的文件你可以自定义是.rar还是.zip
             */
            File file = new File(filePath + "原始数据包.zip");
            if (!file.exists()) {
                file.createNewFile();
            }
            response.reset();
            // response.getWriter()
            // 创建文件输出流
            FileOutputStream fous = new FileOutputStream(file);
            /**
             * 打包的方法我们会用到ZipOutputStream这样一个输出流, 所以这里我们把输出流转换一下
             */
            ZipOutputStream zipOut = new ZipOutputStream(fous);

            /**
             * 这个方法接受的就是一个所要打包文件的集合， 还有一个ZipOutputStream
             */
            zipFile(files, zipOut);
            zipOut.close();
            fous.close();
            return downloadZip(file, response);
        } catch (Exception e) {
            e.printStackTrace();
        }
        /**
         * 直到文件的打包已经成功了， 文件的打包过程被我封装在FileUtil.zipFile这个静态方法中，
         * 稍后会呈现出来，接下来的就是往客户端写数据了
         */

        return response;
    }

    /**
     * 把接受的全部文件打成压缩包
     *
     * @param files        files
     * @param outputStream outputStream
     */
    @SuppressWarnings("rawtypes")
    public static void zipFile(List files, ZipOutputStream outputStream) {
        int size = files.size();
        for (int i = 0; i < size; i++) {
            File file = (File) files.get(i);
            zipFile(file, outputStream);
        }
    }

    public static HttpServletResponse downloadZip(File file,
                                                  HttpServletResponse response) {
        try {
            // 以流的形式下载文件。
            InputStream fis = new BufferedInputStream(new FileInputStream(
                    file.getPath()));
            byte[] buffer = new byte[fis.available()];
            fis.read(buffer);
            fis.close();
            // 清空response
            response.reset();

            OutputStream toClient = new BufferedOutputStream(
                    response.getOutputStream());
            response.setContentType("application/octet-stream");

            // 如果输出的是中文名的文件，在此处就要用URLEncoder.encode方法进行处理
            response.setHeader("Content-Disposition", "attachment;filename="
                    + URLEncoder.encode(file.getName(), "UTF-8"));
            toClient.write(buffer);
            toClient.flush();
            toClient.close();
        } catch (IOException ex) {
            ex.printStackTrace();
        } finally {
            try {
                File f = new File(file.getPath());
                f.delete();
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        return response;
    }

    /**
     * 根据输入的文件与输出流对文件进行打包
     *
     * @param inputFile   inputFile
     * @param ouputStream ouputStream
     */
    public static void zipFile(File inputFile, ZipOutputStream ouputStream) {
        try {
            if (inputFile.exists()) {
                /**
                 * 如果是目录的话这里是不采取操作的， 至于目录的打包正在研究中
                 */
                if (inputFile.isFile()) {
                    FileInputStream in = new FileInputStream(inputFile);
                    BufferedInputStream bins = new BufferedInputStream(in, 512);
                    // org.apache.tools.zip.ZipEntry
                    ZipEntry entry = new ZipEntry(inputFile.getName());
//                    entry.setUnixMode(644);
//                    ouputStream.setEncoding("GBK");
                    ouputStream.putNextEntry(entry);
                    // 向压缩文件中输出数据
                    int nNumber;
                    byte[] buffer = new byte[512];
                    while ((nNumber = bins.read(buffer)) != -1) {
                        ouputStream.write(buffer, 0, nNumber);
                    }
                    // 关闭创建的流对象
                    bins.close();
                    in.close();
                } else {
                    try {
                        File[] files = inputFile.listFiles();
                        for (int i = 0; i < files.length; i++) {
                            zipFile(files[i], ouputStream);
                        }
                    } catch (Exception e) {
                        e.printStackTrace();
                    }
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

}


